package com.hmall.gateway.filter;

import org.apache.commons.lang.StringUtils;
import org.springframework.cloud.gateway.filter.GatewayFilterChain;
import org.springframework.cloud.gateway.filter.GlobalFilter;
import org.springframework.core.Ordered;
import org.springframework.core.annotation.Order;
import org.springframework.http.HttpStatus;
import org.springframework.http.server.reactive.ServerHttpRequest;
import org.springframework.http.server.reactive.ServerHttpResponse;
import org.springframework.stereotype.Component;
import org.springframework.util.MultiValueMap;
import org.springframework.web.server.ServerWebExchange;
import reactor.core.publisher.Mono;

/**
 * @Author: ahcfl
 * @Date: 2021年07月30日 23:49
 * @Description:
 */
public class LoginFilter implements GlobalFilter, Ordered {
    /**
     * 处理当前请求，有必要的话通过@link GatewayFilterChain}将请求交给下一个过滤器处理
     * 每一个过滤器都必须指定一个int类型的order值，**order值越小，优先级越高，执行顺序越靠前**。
     *
     * @param exchange 请求上下文，里面可以获取Request、Response等信息
     * @param chain    用来把请求委托给下一个过滤器
     * @return {@code Mono<Void>} 返回标示当前过滤器业务结束
     */
    @Override
    public Mono<Void> filter(ServerWebExchange exchange, GatewayFilterChain chain) {
        ServerHttpRequest request = exchange.getRequest();
        // 获取请求参数 authorization
        MultiValueMap<String, String> queryParams = request.getQueryParams();

        // 注意：这里我们所有经过网关的请求，都在请求头中添加一个头信息：authorization = 2
        String authorization = queryParams.getFirst("authorization");
        if (StringUtils.isNotBlank(authorization)) {
            int parseInt = Integer.parseInt(authorization);
            if (parseInt == 2) {
                // 包含请求头信息的 放行到下一个过滤器
                return chain.filter(exchange);
            }
        }

        // 不包含请求头信息, 用户没有权限
        ServerHttpResponse response = exchange.getResponse();
        response.setStatusCode(HttpStatus.UNAUTHORIZED);
        return response.setComplete();  // 返回null
    }

    // 过滤器优先级 order值越小，优先级越高
    @Override
    public int getOrder() {
        return 0;
    }
}
